#include "DataPort.h"

DataPort::DataPort(BaseNode* parentNode, PortType type, const QString& dataType)
    : BasePort(parentNode, type)
    , m_dataType(dataType)
{
    // 根据数据类型设置不同的颜色
    if (dataType == "number") {
        m_color = QColor(100, 180, 255);  // 蓝色表示数字
    } else if (dataType == "boolean") {
        m_color = QColor(255, 100, 100);  // 红色表示布尔值
    } else if (dataType == "string") {
        m_color = QColor(100, 255, 150);  // 绿色表示字符串
    } else {
        m_color = QColor(200, 200, 100);  // 黄色表示通用数据
    }
}

QString DataPort::getDataType() const
{
    return m_dataType;
}

QColor DataPort::getPortColor() const
{
    return m_color;
}